package k2_oop.p9_collection.demo16;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 给定一个整数数组 nums 和一个整数目标值 target，
 * 请你在该数组中找出 和为目标值 target 的那两个整数，并返回它们的数组下标
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
 * 你可以按任意顺序返回答案
 * 进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？（选做）
 */
public class TwoNumSum {

    /**
     *
     * @param nums
     * @param target
     * @return
     */
    public int[] method1(int[] nums,int target){
        for (int i = 0; i < nums.length-1; i++) {
            for (int j = i+1; j < nums.length; j++) {
                if (nums[i]+nums[j]==target) {
                    return new int[]{i,j};
                }
            }
        }
        return null;
    }

    /**
     * 空间换时间
     * @param nums
     * @param target
     * @return
     */
    public int[] method2(int[] nums,int target){

        //以数字为key，以此数字对应的索引为value
        Map<Integer,Integer> numIndexMap = new HashMap<>();

        for (int i = 0; i < nums.length; i++) {
            //去Map中查找有没有能有当前数值相加等于目标值的元素
            if(numIndexMap.containsKey(target-nums[i])){
                return new int[]{i,numIndexMap.get(target-nums[i])};
            }
            numIndexMap.put(nums[i],i);
        }
        return null;

    }

    public static void main(String[] args) {

        TwoNumSum twoNumSum = new TwoNumSum();

        int[] nums = {3,2,4};
        int target = 6;

        int[] indexs = twoNumSum.method2(nums,target);
        System.out.println(Arrays.toString(indexs));


    }
}
